Fix imports

smithy-lang · Nov 4, 2024 · 88c163e · 88c163e
1 parent 878687e
commit 88c163e
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/...-codegen/src/main/java/software/amazon/polymorph/smithygo/codegen/StructureGenerator.java b/...-codegen/src/main/java/software/amazon/polymorph/smithygo/codegen/StructureGenerator.java
@@ -122,20 +122,18 @@ public void renderStructure(Runnable runnable, boolean isInputStructure) {
             memberSymbol.getProperty("Referred", Symbol.class).get();
           var refShape = targetShape.expectTrait(ReferenceTrait.class);
           if (refShape.isService()) {
+            namespace =
+              SmithyNameResolver.shapeNamespace(
+                model.expectShape(refShape.getReferentId())
+              );
             if (
               model
                 .expectShape(refShape.getReferentId())
                 .hasTrait(ServiceTrait.class)
             ) {
-              namespace =
-                DafnyNameResolver.dafnyTypesNamespace(
-                  model.expectShape(refShape.getReferentId())
-                );
-            } else {
-              namespace =
-                SmithyNameResolver.shapeNamespace(
-                  model.expectShape(refShape.getReferentId())
-                );
+              writer.addImport(SmithyNameResolver.getGoModuleNameForSdkNamespace(
+                refShape.getReferentId().getNamespace()
+              ));
             }
           }
           if (