diff --git a/client/driver/driver_mock.go b/client/driver/driver_mock.go
deleted file mode 100644
index c0db54e85..000000000
--- a/client/driver/driver_mock.go
+++ /dev/null
@@ -1,8 +0,0 @@
-// +build nomad_test
-
-package driver
-
-// Add the mock driver
-func init() {
-	BuiltinDrivers["mock_driver"] = NewMockDriver
-}
diff --git a/client/driver/mock_driver.go b/client/driver/mock_driver.go
index bc1004683..4e46a3997 100644
--- a/client/driver/mock_driver.go
+++ b/client/driver/mock_driver.go
@@ -16,6 +16,11 @@ import (
 	"github.com/hashicorp/nomad/nomad/structs"
 )
 
+// Add the mock driver to the list of builtin drivers
+func init() {
+	BuiltinDrivers["mock_driver"] = NewMockDriver
+}
+
 // MockDriverConfig is the driver configuration for the MockDriver
 type MockDriverConfig struct {