diff --git a/Dockerfile b/Dockerfile
index da681f5..96659c1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -49,4 +49,4 @@ WORKDIR /app
 
 # Install go dependencies, build the wasm module, push it to the registry
 #CMD ["sh", "-c", "go mod download && go mod verify && go mod tidy && wash build && wash push $REGISTRY build/*_s.wasm"]
-CMD ["sh", "-c", "go env -w GOFLAGS=-buildvcs=false && go mod download && go mod verify && wash build && wash push $REGISTRY build/*_s.wasm"]
\ No newline at end of file
+CMD ["sh", "-c", "go env -w GOFLAGS=-buildvcs=false && go mod download && go mod verify && wash build && wash push $REGISTRY build/*.wasm"]
\ No newline at end of file
diff --git a/go.mod b/go.mod
index 8a418e5..e967911 100644
--- a/go.mod
+++ b/go.mod
@@ -1,25 +1,25 @@
-module gitea.rebus.ninja/lore/wasm-nats-producer-client
-
-go 1.23.0
-
-require (
-	github.com/bytecodealliance/wasm-tools-go v0.3.2
-	go.wasmcloud.dev/component v0.0.5
-	go.wasmcloud.dev/wadge v0.7.0
-)
-
-require (
-	github.com/coreos/go-semver v0.3.1 // indirect
-	github.com/docker/libtrust v0.0.0-20160708172513-aabc10ec26b7 // indirect
-	github.com/klauspost/compress v1.17.11 // indirect
-	github.com/opencontainers/go-digest v1.0.0 // indirect
-	github.com/regclient/regclient v0.8.0 // indirect
-	github.com/sirupsen/logrus v1.9.3 // indirect
-	github.com/ulikunitz/xz v0.5.12 // indirect
-	github.com/urfave/cli/v3 v3.0.0-beta1 // indirect
-	go.bytecodealliance.org v0.4.1 // indirect
-	golang.org/x/mod v0.22.0 // indirect
-	golang.org/x/sync v0.10.0 // indirect
-	golang.org/x/sys v0.28.0 // indirect
-	golang.org/x/tools v0.28.0 // indirect
-)
+module gitea.rebus.ninja/lore/wasm-nats-producer-client
+
+go 1.23.0
+
+require (
+	github.com/bytecodealliance/wasm-tools-go v0.3.2
+	go.wasmcloud.dev/component v0.0.5
+	go.wasmcloud.dev/wadge v0.7.0
+)
+
+require (
+	github.com/coreos/go-semver v0.3.1 // indirect
+	github.com/docker/libtrust v0.0.0-20160708172513-aabc10ec26b7 // indirect
+	github.com/klauspost/compress v1.17.11 // indirect
+	github.com/opencontainers/go-digest v1.0.0 // indirect
+	github.com/regclient/regclient v0.8.0 // indirect
+	github.com/sirupsen/logrus v1.9.3 // indirect
+	github.com/ulikunitz/xz v0.5.12 // indirect
+	github.com/urfave/cli/v3 v3.0.0-beta1 // indirect
+	go.bytecodealliance.org v0.4.1 // indirect
+	golang.org/x/mod v0.22.0 // indirect
+	golang.org/x/sync v0.10.0 // indirect
+	golang.org/x/sys v0.28.0 // indirect
+	golang.org/x/tools v0.28.0 // indirect
+)
diff --git a/main.go b/main.go
index f6d616d..36f40d3 100644
--- a/main.go
+++ b/main.go
@@ -2,7 +2,6 @@
 package main
 
 import (
-	"gitea.rebus.ninja/lore/wasm-nats-producer-client/gen/wasmcloud/messaging/handler"
 )
 
 func init() {