diff --git a/ANN_Data_Generator.py b/ANN_Data_Generator.py
index 1a799fb5e2a24957356ee01fa52deb6dd0351ace..14dbed12f0a1206b330666ab8fe2d8cddb5a7c3a 100644
--- a/ANN_Data_Generator.py
+++ b/ANN_Data_Generator.py
@@ -165,7 +165,7 @@ class TrainingDataGenerator(object):
         troubled_indicator = 'without' if is_smooth else 'with'
         print('Calculating data ' + troubled_indicator + ' troubled cells...')
         print('Samples to complete:', num_samples)
-        tic = timeit.default_timer()
+        tic = time.perf_counter()
 
         input_data = np.zeros((num_samples, self._stencil_length+2))
         num_init_cond = len(initial_conditions)
@@ -198,7 +198,7 @@ class TrainingDataGenerator(object):
             if count % 1000 == 0:
                 print(str(count) + ' samples completed.')
 
-        toc = timeit.default_timer()
+        toc = time.perf_counter()
         print('Finished calculating data ' + troubled_indicator + ' troubled cells!')
         print('Calculation time:', toc-tic, '\n')