diff --git a/haversine_02/src/repetition_testing/reptester.cpp b/haversine_02/src/repetition_testing/reptester.cpp
index 6a66181..56efb3f 100644
--- a/haversine_02/src/repetition_testing/reptester.cpp
+++ b/haversine_02/src/repetition_testing/reptester.cpp
@@ -43,7 +43,10 @@ void run_func_test(reptester *tester, reptest_func func, const char *func_name,
   tester->total = 0;
   tester->results = {};
 
-  if (type == ALLOC_TYPE_WITHOUT_MALLOC && !(tester->params.buffer)) {
+  char *buffer = NULL;
+
+  if (type == ALLOC_TYPE_WITH_MALLOC) {
+    buffer = tester->params.buffer;
     tester->params.buffer = (char *)malloc(tester->params.read_size + 1);
     memset(tester->params.buffer, 0, tester->params.read_size + 1);
   }
@@ -75,9 +78,9 @@ void run_func_test(reptester *tester, reptest_func func, const char *func_name,
     ++(tester->current_run);
   }
 
-  if (type == ALLOC_TYPE_WITHOUT_MALLOC && tester->params.buffer) {
+  if (type == ALLOC_TYPE_WITH_MALLOC) {
     free(tester->params.buffer);
-    tester->params.buffer = NULL;
+    tester->params.buffer = buffer;
   }
 
   print_results(tester, func_name);