diff --git a/consumer.go b/consumer.go
index 44e690d..e65a08d 100644
--- a/consumer.go
+++ b/consumer.go
@@ -12,7 +12,7 @@ import (
 	"github.com/aws/aws-sdk-go/service/kinesis"
 	"github.com/aws/aws-sdk-go/service/kinesis/kinesisiface"
 
-	"github.com/opentracing/opentracing-go"
+	opentracing "github.com/opentracing/opentracing-go"
 	"github.com/opentracing/opentracing-go/ext"
 )
 
diff --git a/consumer_test.go b/consumer_test.go
index 5eb8a9e..0765dc1 100644
--- a/consumer_test.go
+++ b/consumer_test.go
@@ -69,7 +69,7 @@ func TestConsumer_Scan(t *testing.T) {
 
 	var resultData string
 	var fnCallCounter int
-	var fn = func(r *Record) ScanStatus {
+	var fn = func(ctx context.Context, r *Record) ScanStatus {
 		fnCallCounter++
 		resultData += string(r.Data)
 		return ScanStatus{}
@@ -121,7 +121,7 @@ func TestConsumer_Scan_NoShardsAvailable(t *testing.T) {
 	}
 
 	var fnCallCounter int
-	var fn = func(r *Record) ScanStatus {
+	var fn = func(ctx context.Context, r *Record) ScanStatus {
 		fnCallCounter++
 		return ScanStatus{}
 	}
@@ -185,7 +185,7 @@ func TestScanShard(t *testing.T) {
 
 	// callback fn appends record data
 	var resultData string
-	var fn = func(r *Record) ScanStatus {
+	var fn = func(ctx context.Context, r *Record) ScanStatus {
 		resultData += string(r.Data)
 		return ScanStatus{}
 	}
@@ -246,7 +246,7 @@ func TestScanShard_StopScan(t *testing.T) {
 
 	// callback fn appends record data
 	var resultData string
-	var fn = func(r *Record) ScanStatus {
+	var fn = func(ctx context.Context, r *Record) ScanStatus {
 		resultData += string(r.Data)
 		return ScanStatus{StopScan: true}
 	}
@@ -281,7 +281,7 @@ func TestScanShard_ShardIsClosed(t *testing.T) {
 		t.Fatalf("new consumer error: %v", err)
 	}
 
-	var fn = func(r *Record) ScanStatus {
+	var fn = func(ctx context.Context, r *Record) ScanStatus {
 		return ScanStatus{}
 	}