diff --git a/codegen/sdk-codegen/build.gradle.kts b/codegen/sdk-codegen/build.gradle.kts
index 57eb849925781..0c7213a8f52fe 100644
--- a/codegen/sdk-codegen/build.gradle.kts
+++ b/codegen/sdk-codegen/build.gradle.kts
@@ -46,6 +46,7 @@ dependencies {
     val smithyVersion: String by project
     implementation(project(":smithy-aws-typescript-codegen"))
     implementation("software.amazon.smithy:smithy-smoke-test-traits:$smithyVersion")
+    implementation("software.amazon.smithy:smithy-aws-smoke-test-model:$smithyVersion")
 }
 
 // This project doesn't produce a JAR.