fix: LibInitializeGuard

axieinfinity · Oct 10, 2024 · ac0327d · ac0327d
1 parent 824adfe
commit ac0327d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/script/libraries/LibInitializeGuard.sol b/script/libraries/LibInitializeGuard.sol
@@ -340,9 +340,12 @@ library LibInitializeGuard {
     string memory path
   ) internal pure returns (string memory contractName) {
     uint256 length = bytes(path).length;
-    // Remove ".sol"
+    contractName = path;
     if (path.endsWith(".sol")) contractName = path.slice(0, length - 4);
-    return string(contractName);
+    string[] memory parts = contractName.split(":");
+    if (parts.length != 0) contractName = parts[parts.length - 1];
+    parts = contractName.split("/");
+    if (parts.length != 0) contractName = parts[parts.length - 1];
   }
 
   /**