diff --git a/spec/amplitude/storage_spec.rb b/spec/amplitude/storage_spec.rb
index 90b4d07..25b8924 100644
--- a/spec/amplitude/storage_spec.rb
+++ b/spec/amplitude/storage_spec.rb
@@ -4,9 +4,9 @@ module AmplitudeAnalytics
       @provider = InMemoryStorageProvider.new
       @storage = @provider.storage
       @workers = Workers.new
-      config = Config.new
-      @storage.setup(config, @workers)
-      @workers.setup(config, @storage)
+      @config = Config.new
+      @storage.setup(@config, @workers)
+      @workers.setup(@config, @storage)
     end
 
     it 'checks if storage is empty after pull' do
@@ -68,6 +68,7 @@ module AmplitudeAnalytics
 
     it 'exceeds max capacity and fails' do
       push_event(@storage, Set.new, MAX_BUFFER_CAPACITY)
+      sleep(1)
       expect(@storage.total_events).to eq(MAX_BUFFER_CAPACITY)
 
       event = BaseEvent.new('test_event', user_id: 'test_user')
diff --git a/spec/amplitude/worker_spec.rb b/spec/amplitude/worker_spec.rb
index 410db7b..ec26183 100644
--- a/spec/amplitude/worker_spec.rb
+++ b/spec/amplitude/worker_spec.rb
@@ -257,7 +257,6 @@ module AmplitudeAnalytics
         while @workers.storage.total_events > 0
           sleep(0.1)
         end
-        sleep(5)
         total_events = @events_dict.values.sum(&:length)
         expect(@workers.storage.total_events).to eq(0)
         expect(total_events).to eq(5000)