diff --git a/sharding/mock/enableEpochsHandlerMock.go b/sharding/mock/enableEpochsHandlerMock.go
index e67f7f486b..93df39fcd2 100644
--- a/sharding/mock/enableEpochsHandlerMock.go
+++ b/sharding/mock/enableEpochsHandlerMock.go
@@ -618,6 +618,11 @@ func (mock *EnableEpochsHandlerMock) IsAutoBalanceDataTriesEnabled() bool {
 	return false
 }
 
+// IsMigrateDataTrieEnabled -
+func (mock *EnableEpochsHandlerMock) IsMigrateDataTrieEnabled() bool {
+	return false
+}
+
 // FixDelegationChangeOwnerOnAccountEnabled -
 func (mock *EnableEpochsHandlerMock) FixDelegationChangeOwnerOnAccountEnabled() bool {
 	return false