diff --git a/tardis/tests/integration_tests/test_integration.py b/tardis/tests/integration_tests/test_integration.py
index 61c07516c31..2733ec6913f 100644
--- a/tardis/tests/integration_tests/test_integration.py
+++ b/tardis/tests/integration_tests/test_integration.py
@@ -135,8 +135,7 @@ def setup(self, request, reference, data_path, pytestconfig):
                 pytest.skip(
                     'Reference data {0} does exist and tests will not '
                     'proceed generating new data'.format(ref_data_path))
-            self.result.to_hdf(path_or_buf=ref_data_path,
-                               suffix_count=False)
+            self.result.to_hdf(file_path=ref_data_path)
             pytest.skip("Reference data saved at {0}".format(
                 data_path['gen_ref_path']
             ))