diff --git a/tests/unit/Get-ComputerInfo.Tests.ps1 b/tests/unit/Get-ComputerInfo.Tests.ps1
index 5c4b9e3..107a0ef 100644
--- a/tests/unit/Get-ComputerInfo.Tests.ps1
+++ b/tests/unit/Get-ComputerInfo.Tests.ps1
@@ -1,5 +1,5 @@
 BeforeAll {
-    Import-Module ../../src/linuxinfo/linuxinfo.psd1 -Function Get-ComputerInfo -Verbose
+    ../../build.ps1 -Force
 }
 
 Describe "Get-ComputerInfo" {
diff --git a/tests/unit/Get-SystemUptime.Tests.ps1 b/tests/unit/Get-SystemUptime.Tests.ps1
index 627e9cb..7924660 100644
--- a/tests/unit/Get-SystemUptime.Tests.ps1
+++ b/tests/unit/Get-SystemUptime.Tests.ps1
@@ -1,5 +1,5 @@
 BeforeAll {
-    Import-Module ../../src/linuxinfo/linuxinfo.psd1 -Function Get-SystemUpTime -Verbose
+    ../../build.ps1 -Force
 }
 
 Describe "Get-SystemUptime" {