diff --git a/Tests/Public/Get-ZertoDatastore.Tests.ps1 b/Tests/Public/Get-ZertoDatastore.Tests.ps1
index 46c470b..6727a99 100644
--- a/Tests/Public/Get-ZertoDatastore.Tests.ps1
+++ b/Tests/Public/Get-ZertoDatastore.Tests.ps1
@@ -7,8 +7,8 @@ Describe $global:function -Tag 'Unit', 'Source', 'Built' {
     Context "$global:function::Parameter Unit Tests" {
 
         It "has a non-mandatory string parameter for the datacenterIdentifier" {
-            Get-Command $file.BaseName | Should -HaveParameter datastoreIdentifier
-            Get-Command $file.BaseName | Should -HaveParameter datastoreIdentifier -Type String[]
+            Get-Command $global:function | Should -HaveParameter datastoreIdentifier
+            Get-Command $global:function | Should -HaveParameter datastoreIdentifier -Type String[]
         }
     }
 
diff --git a/Tests/Public/Get-ZertoLocalSite.Tests.ps1 b/Tests/Public/Get-ZertoLocalSite.Tests.ps1
index 4412139..4fea226 100644
--- a/Tests/Public/Get-ZertoLocalSite.Tests.ps1
+++ b/Tests/Public/Get-ZertoLocalSite.Tests.ps1
@@ -6,8 +6,8 @@ Describe $global:function -Tag 'Unit', 'Source', 'Built' {
 
     Context "$global:function::Parameter Unit Tests" {
         it "Has a non-mandatory switch parameter for the pairing Statuses" {
-            Get-Command $file.BaseName | Should -HaveParameter pairingstatuses
-            Get-Command $file.BaseName | Should -HaveParameter pairingstatuses -Type switch
+            Get-Command $global:function | Should -HaveParameter pairingstatuses
+            Get-Command $global:function | Should -HaveParameter pairingstatuses -Type switch
         }
     }