diff --git a/packages/@aws-cdk-testing/cli-integ/tests/init-go/init-go.integtest.ts b/packages/@aws-cdk-testing/cli-integ/tests/init-go/init-go.integtest.ts
index f9772a7d80460..779078ac68739 100644
--- a/packages/@aws-cdk-testing/cli-integ/tests/init-go/init-go.integtest.ts
+++ b/packages/@aws-cdk-testing/cli-integ/tests/init-go/init-go.integtest.ts
@@ -8,7 +8,7 @@ import { integTest, withTemporaryDirectory, ShellHelper, withPackages } from '..
     await context.packages.makeCliAvailable();
 
     await shell.shell(['cdk', 'init', '-l', 'go', template]);
-    await shell.shell(['go', 'mod', 'edit', '-replace', 'github.com/aws/aws-cdk-go/awscdk=$dist_root/go/awscdk']);
+    await shell.shell(['go', 'mod', 'edit', '-replace', 'github.com/aws/aws-cdk-go/awscdk/v2=$CODEBUILD_SRC_DIR/go/awscdk']);
     await shell.shell(['go', 'mod', 'tidy']);
     await shell.shell(['go', 'test']);
     await shell.shell(['cdk', 'synth']);
diff --git a/packages/aws-cdk/lib/init-templates/sample-app/go/%name%_test.template.go b/packages/aws-cdk/lib/init-templates/sample-app/go/%name%_test.template.go
index 56181ebe4b377..d7d6ce936326d 100644
--- a/packages/aws-cdk/lib/init-templates/sample-app/go/%name%_test.template.go
+++ b/packages/aws-cdk/lib/init-templates/sample-app/go/%name%_test.template.go
@@ -16,7 +16,7 @@ func Test%name.PascalCased%Stack(t *testing.T) {
 	stack := New%name.PascalCased%Stack(app, "MyStack", nil)
 
 	// THEN
-	template := assertions.Template_FromStack(stack)
+	template := assertions.Template_FromStack(stack, nil)
 
 	template.HasResourceProperties(jsii.String("AWS::SQS::Queue"), map[string]interface{}{
 		"VisibilityTimeout": 300,