diff --git a/generate.go b/generate.go
index 43d2efd..95d8d5d 100644
--- a/generate.go
+++ b/generate.go
@@ -1,7 +1,6 @@
 package tally
 
 import (
+	//go:generate mockgen -package tallymock -destination tallymock/stats_reporter.go -imports github.com/uber-go/tally github.com/uber-go/tally StatsReporter
 	_ "github.com/golang/mock/mockgen/model"
 )
-
-//go:generate mockgen -package tallymock -destination tallymock/stats_reporter.go -imports github.com/uber-go/tally github.com/uber-go/tally StatsReporter