diff --git a/mock_cache_test.go b/mock_cache_test.go
index 4a5fd9ad..d2e02e66 100644
--- a/mock_cache_test.go
+++ b/mock_cache_test.go
@@ -41,19 +41,13 @@ func (m *MockCache) UpdateExpiration(key string, expireAt int64) bool {
 
 func (m *MockCache) GetItem(key string) (value *guber.CacheItem, ok bool) {
 	args := m.Called(key)
-	var retval *guber.CacheItem
-	if retval2, ok := args.Get(0).(*guber.CacheItem); ok {
-		retval = retval2
-	}
+	retval, _ := args.Get(0).(*guber.CacheItem)
 	return retval, args.Bool(1)
 }
 
 func (m *MockCache) Each() chan *guber.CacheItem {
 	args := m.Called()
-	var retval chan *guber.CacheItem
-	if retval2, ok := args.Get(0).(chan *guber.CacheItem); ok {
-		retval = retval2
-	}
+	retval, _ := args.Get(0).(chan *guber.CacheItem)
 	return retval
 }
 
diff --git a/mock_loader_test.go b/mock_loader_test.go
index 220ba5c0..3a52c686 100644
--- a/mock_loader_test.go
+++ b/mock_loader_test.go
@@ -31,10 +31,7 @@ var _ guber.Loader = &MockLoader2{}
 
 func (m *MockLoader2) Load() (chan *guber.CacheItem, error) {
 	args := m.Called()
-	var retval chan *guber.CacheItem
-	if retval2, ok := args.Get(0).(chan *guber.CacheItem); ok {
-		retval = retval2
-	}
+	retval := args.Get(0).(chan *guber.CacheItem)
 	return retval, args.Error(1)
 }
 
diff --git a/mock_store_test.go b/mock_store_test.go
index 1b07ec3f..8d3d1bdd 100644
--- a/mock_store_test.go
+++ b/mock_store_test.go
@@ -37,10 +37,7 @@ func (m *MockStore2) OnChange(ctx context.Context, r *guber.RateLimitReq, item *
 
 func (m *MockStore2) Get(ctx context.Context, r *guber.RateLimitReq) (*guber.CacheItem, bool) {
 	args := m.Called(ctx, r)
-	var retval *guber.CacheItem
-	if retval2, ok := args.Get(0).(*guber.CacheItem); ok {
-		retval = retval2
-	}
+	retval, _ := args.Get(0).(*guber.CacheItem)
 	return retval, args.Bool(1)
 }