diff --git a/trove/tests/unittests/guestagent/test_mysql_manager.py b/trove/tests/unittests/guestagent/test_mysql_manager.py
index 308f51c13b..507979aa4c 100644
--- a/trove/tests/unittests/guestagent/test_mysql_manager.py
+++ b/trove/tests/unittests/guestagent/test_mysql_manager.py
@@ -214,6 +214,8 @@ class GuestAgentManagerTest(trove_testtools.TestCase):
         snapshot = {'replication_strategy': self.replication_strategy,
                     'dataset': {'dataset_size': 1.0},
                     'config': None}
+        total_size = snapshot['dataset']['dataset_size'] + 1
+        self.mock_gfvs_class.return_value = {'total': total_size}
         self._prepare_dynamic(snapshot=snapshot)
 
     @patch.multiple(dbaas.MySqlAdmin,