diff --git a/contracts/BEBase.sol b/contracts/BEBase.sol
index 2cbafd0..0790d6b 100644
--- a/contracts/BEBase.sol
+++ b/contracts/BEBase.sol
@@ -27,7 +27,7 @@ abstract contract BEBase is ERC721, AccessControlEnumerable, ERC721Enumerable, O
 
         string memory baseURI = _baseURI();
         uint256 tokenSub = tokenId % 1000;
-        return bytes(baseURI).length > 0 ? string(abi.encodePacked(baseURI, tokenSub.toString())) : "";
+        return bytes(baseURI).length > 0 ? string(abi.encodePacked(baseURI, tokenSub.toString(), '.png')) : "";
     }
 
     /**
diff --git a/contracts/Migrations.sol b/contracts/Migrations.sol
index 2612207..4a77984 100644
--- a/contracts/Migrations.sol
+++ b/contracts/Migrations.sol
@@ -16,4 +16,9 @@ contract Migrations {
   function setCompleted(uint completed) external restricted {
     last_completed_migration = completed;
   }
+
+  function upgrade(address new_address) external restricted {
+    Migrations upgraded = Migrations(new_address);
+    upgraded.setCompleted(last_completed_migration);
+  }
 }