diff --git a/src/main/main.c b/src/main/main.c
index ab54c20845aaac44fd54f539341929b38759f381..3f3d8396418342a23acb5e31a4998869a480fdeb 100644
--- a/src/main/main.c
+++ b/src/main/main.c
@@ -218,13 +218,13 @@ void simulate_master_read_and_scatter(int dims, size_t n, global_context_t *ctx)
 
     if(I_AM_MASTER && ctx -> world_size <= 6)
     {
-        test_file_path(OUT_DATA);
-        test_file_path(OUT_CLUSTER_ASSIGN);
+        test_file_path(ctx -> output_data_file);
+        test_file_path(ctx -> output_assignment_file);
     }
     else
     {
-        test_distributed_file_path(ctx, OUT_DATA);
-        test_distributed_file_path(ctx, OUT_CLUSTER_ASSIGN);
+        test_distributed_file_path(ctx, ctx -> output_data_file);
+        test_distributed_file_path(ctx, ctx -> output_assignment_file);
     }