diff --git a/Dockerfile b/Dockerfile
index dee8954..0f223f9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,7 +12,8 @@ COPY go.sum go.sum
 RUN go mod download
 
 # Copy the go source
-COPY cmd/ cmd/
+COPY cmd/main.go cmd/main.go
+COPY cmd/logging.go cmd/logging.go
 COPY api/ api/
 COPY internal/ internal/
 
diff --git a/internal/controller/locustcluster_controller.go b/internal/controller/locustcluster_controller.go
index be56a7c..eda1842 100644
--- a/internal/controller/locustcluster_controller.go
+++ b/internal/controller/locustcluster_controller.go
@@ -18,6 +18,8 @@ package controller
 
 import (
 	"context"
+
+	locustiov1alpha1 "git.lilpenguins.com/crichardson/locust-operator/api/v1alpha1"
 	apps "k8s.io/api/apps/v1"
 	batch "k8s.io/api/batch/v1"
 	apierrors "k8s.io/apimachinery/pkg/api/errors"
@@ -28,8 +30,6 @@ import (
 	"sigs.k8s.io/controller-runtime/pkg/event"
 	logf "sigs.k8s.io/controller-runtime/pkg/log"
 	"sigs.k8s.io/controller-runtime/pkg/predicate"
-
-	locustiov1alpha1 "git.lilpenguins.com/crichardson/locust-operator/api/v1alpha1"
 )
 
 // LocustClusterReconciler reconciles a LocustCluster object