diff --git a/cmd/service/service_test.go b/cmd/service/service_test.go
index 9261a4f..7d884de 100644
--- a/cmd/service/service_test.go
+++ b/cmd/service/service_test.go
@@ -60,7 +60,7 @@ func Test_start(t *testing.T) {
 			CLI.Port = tt.args.port
 			CLI.Bucket = tt.args.bucket
 			CLI.ReturnURL = tt.args.url
-			if err := start(logger.Logger); (err != nil) != tt.wantErr {
+			if err := start(logger); (err != nil) != tt.wantErr {
 				t.Errorf("start() error = %v, wantErr %v", err, tt.wantErr)
 			}
 			logger.Check(t, tt.wantLogged)
diff --git a/go.mod b/go.mod
index c8753b3..d4173e3 100644
--- a/go.mod
+++ b/go.mod
@@ -7,7 +7,7 @@ require (
 	github.com/apex/log v1.9.0
 	github.com/aws/aws-sdk-go v1.44.6
 	github.com/stretchr/testify v1.7.2
-	gitlab.com/unboundsoftware/apex-mocks v0.1.0
+	gitlab.com/unboundsoftware/apex-mocks v0.2.0
 )
 
 require (
diff --git a/go.sum b/go.sum
index 5c3d6c6..bf65157 100644
--- a/go.sum
+++ b/go.sum
@@ -62,8 +62,8 @@ github.com/tj/go-buffer v1.1.0/go.mod h1:iyiJpfFcR2B9sXu7KvjbT9fpM4mOelRSDTbntVj
 github.com/tj/go-elastic v0.0.0-20171221160941-36157cbbebc2/go.mod h1:WjeM0Oo1eNAjXGDx2yma7uG2XoyRZTq1uv3M/o7imD0=
 github.com/tj/go-kinesis v0.0.0-20171128231115-08b17f58cb1b/go.mod h1:/yhzCV0xPfx6jb1bBgRFjl5lytqVqZXEaeqWP8lTEao=
 github.com/tj/go-spin v1.1.0/go.mod h1:Mg1mzmePZm4dva8Qz60H2lHwmJ2loum4VIrLgVnKwh4=
-gitlab.com/unboundsoftware/apex-mocks v0.1.0 h1:mIKtXumBzKh2T/hOHiCuT+Jx3Qk3uf3XzhQxesS4ELE=
-gitlab.com/unboundsoftware/apex-mocks v0.1.0/go.mod h1:l4opqnCT3PJEGTorga2TVzusUybgKhKmfxAHu/2iLJU=
+gitlab.com/unboundsoftware/apex-mocks v0.2.0 h1:IFt+uyIoOkSl4qdUBLUSIvOhaRdQRGB6TnpZqfRuXqY=
+gitlab.com/unboundsoftware/apex-mocks v0.2.0/go.mod h1:FGsQjCu/nS6b+QaBpAFvms6p0Chr0aobGcUPeeZNSNo=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20190426145343-a29dc8fdc734/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
 golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
diff --git a/server/server_test.go b/server/server_test.go
index 45a61f6..287eea3 100644
--- a/server/server_test.go
+++ b/server/server_test.go
@@ -226,7 +226,7 @@ func TestServer(t *testing.T) {
 			if tt.args.store != nil {
 				store = tt.args.store(t)
 			}
-			server := New(store, tt.args.url, logger.Logger)
+			server := New(store, tt.args.url, logger)
 			server.(*Server).now = func() time.Time {
 				return time.Date(2021, 11, 25, 7, 43, 12, 0, time.UTC)
 			}