diff --git a/tests/scripts/generate_code.pl b/tests/scripts/generate_code.pl
index d167605..40f993f 100755
--- a/tests/scripts/generate_code.pl
+++ b/tests/scripts/generate_code.pl
@@ -65,10 +65,11 @@
 
 $test_helpers
 
-$suite_pre_code
-
 END
 
+$test_main =~ s/SUITE_PRE_DEP/$suite_pre_code/;
+$test_main =~ s/SUITE_POST_DEP/$suite_post_code/;
+
 while($test_cases =~ /\/\* BEGIN_CASE *([\w:]*) \*\/\n(.*?)\n\/\* END_CASE \*\//msg)
 {
     my $function_deps = $1;
@@ -219,8 +220,6 @@
 $test_main =~ s/MAPPING_CODE/$mapping_code/;
 
 print TEST_FILE << "END";
-$suite_post_code
-
 $test_main
 END