diff --git a/locked-nft/contracts/NFTLocker.cdc b/locked-nft/contracts/NFTLocker.cdc
index 08ecfea..d826b21 100644
--- a/locked-nft/contracts/NFTLocker.cdc
+++ b/locked-nft/contracts/NFTLocker.cdc
@@ -79,7 +79,7 @@ access(all) contract NFTLocker {
     /// Determine if NFT can be unlocked
     ///
     access(all) view fun canUnlockToken(id: UInt64, nftType: Type): Bool {
-        if let lockedTokens = NFTLocker.lockedTokens[nftType] {
+        if let lockedTokens = &NFTLocker.lockedTokens[nftType] as &{UInt64: NFTLocker.LockedData}? {
             if let lockedToken = lockedTokens[id] {
                 if lockedToken.lockedUntil <= UInt64(getCurrentBlock().timestamp) {
                     return true
diff --git a/locked-nft/flow.json b/locked-nft/flow.json
index 567612b..ddc7df7 100644
--- a/locked-nft/flow.json
+++ b/locked-nft/flow.json
@@ -20,6 +20,7 @@
 			"source": "./contracts/NFTLocker.cdc",
 			"aliases": {
 				"emulator": "f8d6e0586b0a20c7",
+				"mainnet": "b6f2481eba4df97b",
 				"testnet": "ef4cd3d07a7b43ce"
 			}
 		},
diff --git a/locked-nft/lib/go/test/Makefile b/locked-nft/lib/go/test/Makefile
index b79c5c1..fa2dc33 100644
--- a/locked-nft/lib/go/test/Makefile
+++ b/locked-nft/lib/go/test/Makefile
@@ -1,6 +1,6 @@
 .PHONY: test
 test:
-	go test ./...
+	CGO_ENABLED=0 go test -tags=no_cgo ./...
 
 .PHONY: ci
 ci: test
\ No newline at end of file