diff --git a/volume/drivers/test/driver.go b/volume/drivers/test/driver.go
index 1b19205bd..3b2aa4b3c 100644
--- a/volume/drivers/test/driver.go
+++ b/volume/drivers/test/driver.go
@@ -3,6 +3,7 @@ package test
 import (
 	"context"
 	"fmt"
+	"github.com/libopenstorage/openstorage/pkg/correlation"
 	"os"
 	"os/exec"
 	"path"