diff --git a/consumer/sqsclient_test.go b/consumer/sqsclient_test.go
index afa4b44..0edb022 100644
--- a/consumer/sqsclient_test.go
+++ b/consumer/sqsclient_test.go
@@ -112,7 +112,7 @@ func (ut *UnitTest) TestReceiveMessage() {
 
 	go client.ReceiveMessages("https://fake-queue-url", ch)
 
-	time.Sleep(500 * time.Millisecond)
+	time.Sleep(600 * time.Millisecond)
 
 	fmt.Println(len(ch))
 
@@ -293,7 +293,7 @@ func (uts *UnitTest) TestPoll() {
 
 	go client.Poll()
 
-	time.Sleep(400 * time.Millisecond)
+	time.Sleep(600 * time.Millisecond)
 
 	uts.mockSQSService.AssertCalled(uts.T(), "ReceiveMessage", &sqs.ReceiveMessageInput{
 		QueueUrl:            aws.String("https://fake-queue-url"),
@@ -375,7 +375,7 @@ func (uts *UnitTest) TestPollPrefixBased() {
 
 	go client.Poll()
 
-	time.Sleep(400 * time.Millisecond)
+	time.Sleep(600 * time.Millisecond)
 
 	uts.mockSQSService.AssertCalled(uts.T(), "ListQueues", &sqs.ListQueuesInput{
 		QueueNamePrefix: aws.String("fake-queue-name"),
@@ -401,7 +401,7 @@ func (uts *UnitTest) TestStart() {
 
 	go client.Start()
 
-	time.Sleep(100 * time.Millisecond)
+	time.Sleep(600 * time.Millisecond)
 
 	uts.mockSQSService.AssertCalled(uts.T(), "ReceiveMessage", &sqs.ReceiveMessageInput{
 		QueueUrl:            aws.String("https://fake-queue-url"),
diff --git a/mocks/SQSService.go b/mocks/SQSService.go
index b84537d..879d84d 100644
--- a/mocks/SQSService.go
+++ b/mocks/SQSService.go
@@ -3,6 +3,8 @@
 package mocks
 
 import (
+	"time"
+
 	sqs "github.com/aws/aws-sdk-go/service/sqs"
 	mock "github.com/stretchr/testify/mock"
 )
@@ -118,6 +120,8 @@ func (_m *SQSService) ListQueues(input *sqs.ListQueuesInput) (*sqs.ListQueuesOut
 
 // ReceiveMessage provides a mock function with given fields: input
 func (_m *SQSService) ReceiveMessage(input *sqs.ReceiveMessageInput) (*sqs.ReceiveMessageOutput, error) {
+	time.Sleep(500 * time.Millisecond)
+
 	ret := _m.Called(input)
 
 	var r0 *sqs.ReceiveMessageOutput