fix check

unloggedio · Jun 24, 2024 · 24cd381 · 24cd381
1 parent 4b2bc15
commit 24cd381
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/test/python/Target.py b/src/test/python/Target.py
@@ -6,6 +6,7 @@
 # define constants
 main_method_identifier = "public static void main"
 unlogged_annotation = "@Unlogged(port = " + "12100" + ")"
+unlogged_annotation_base_string = "Unlogged"
 unlogged_import = "import io.unlogged.Unlogged;"
 
 class ReplayTest:
@@ -106,7 +107,7 @@ def modify_main(self):
 			# check annotation
 			annotation_present = False
 			for line in file:
-				if (unlogged_annotation in line):
+				if (unlogged_annotation_base_string in line):
 					annotation_present = True
 
 			# add annotation