diff --git a/lib/trailblazer/core/utils/convert_operation_test.rb b/lib/trailblazer/core/utils/convert_operation_test.rb
index 52420d8..a8585d1 100644
--- a/lib/trailblazer/core/utils/convert_operation_test.rb
+++ b/lib/trailblazer/core/utils/convert_operation_test.rb
@@ -40,14 +40,13 @@ def self.call(filepath, target:, source_gem:)
               # puts "@@@@@ #{line.inspect}"
               line = ""
             else
-              line = line.sub("< Trailblazer::Activity::Railway", "< Trailblazer::Operation")
-              line = line.gsub("::Activity", "::Operation")
-
               if match = line.match(/#!hint (.+)/)
                 ws = line.match(/^(\s+)/) # Find out number of whitespace before code starts.
 
                 line = (" " * ws[1].size) + match[1] + "\n"
               else
+                line = line.sub("< Trailblazer::Activity::Railway", "< Trailblazer::Operation")
+                line = line.gsub("::Activity", "::Operation")
 
               # if within_marker
                 line = line.sub("signal, (ctx, _) =", "result =")