diff --git a/go_templates/client.vm b/go_templates/client.vm
index 6bbbf30..6478a05 100644
--- a/go_templates/client.vm
+++ b/go_templates/client.vm
@@ -8,7 +8,7 @@ import (
   "github.com/algorand/go-algorand-sdk/v2/client/v2/common"
 #foreach( $q in $queries )
 ## TODO: Something like: $str.matches("#oapiToGo($qp)", "model\..*")
-#if ( "$go.queryTypeMapper(${str.capitalize($q.name)})" == "TealDryrun" )
+#if ( "$go.queryTypeMapper(${str.capitalize($q.name)})" == "SimulateTransaction" )
   "github.com/algorand/go-algorand-sdk/v2/client/v2/common/models"
 #end
 #end