diff --git a/packages/hardhat/contracts/Controller.sol b/packages/hardhat/contracts/Controller.sol
index 599ad9a..cdf210a 100644
--- a/packages/hardhat/contracts/Controller.sol
+++ b/packages/hardhat/contracts/Controller.sol
@@ -125,6 +125,9 @@ contract Controller is Ownable {
 
 		// Transfer
 
+		trainingCount[receiver] = 0;
+		trainingTimestamp[receiver] = 0;
+
 		(bool success, ) = receiver.call{ value: reward }("");
 		require(success, "Transfer failed");
 
diff --git a/packages/nextjs/contracts/deployedContracts.ts b/packages/nextjs/contracts/deployedContracts.ts
index 1c0ae22..0775af7 100644
--- a/packages/nextjs/contracts/deployedContracts.ts
+++ b/packages/nextjs/contracts/deployedContracts.ts
@@ -7,7 +7,7 @@ import { GenericContractsDeclaration } from "~~/utils/scaffold-eth/contract";
 const deployedContracts = {
   31337: {
     Controller: {
-      address: "0x5FbDB2315678afecb367f032d93F642f64180aa3",
+      address: "0xa513E6E4b8f2a923D98304ec87F64353C4D5C853",
       abi: [
         {
           inputs: [