Fix imports

smithy-lang · Nov 4, 2024 · 2f7f08b · 2f7f08b
1 parent 830563c
commit 2f7f08b
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/...ftware/amazon/polymorph/smithygo/localservice/shapevisitor/DafnyToSmithyShapeVisitor.java b/...ftware/amazon/polymorph/smithygo/localservice/shapevisitor/DafnyToSmithyShapeVisitor.java
@@ -152,12 +152,11 @@ protected String referenceStructureShape(final StructureShape shape) {
       }
       if (!this.isOptional) {
         if (serviceShape.hasTrait(ServiceTrait.class)) {
-          writer.addImportFromModule(
-            SmithyNameResolver.getGoModuleNameForSmithyNamespace(
-              resourceOrService.toShapeId().getNamespace()
-            ),
-            DafnyNameResolver.dafnyTypesNamespace(serviceShape)
-          );
+          writer.addImport(
+                SmithyNameResolver.getGoModuleNameForSdkNamespace(
+                  serviceShape.getId().getNamespace()
+                )
+              );
           return """
           shim, ok := %1$s.(*%2$swrapped.Shim)
           if !ok {