diff --git a/core/src/main/scala/org/apache/spark/eventhubscommon/client/EventHubsClientWrapper.scala b/core/src/main/scala/org/apache/spark/eventhubscommon/client/EventHubsClientWrapper.scala
index b16bba75..86095204 100644
--- a/core/src/main/scala/org/apache/spark/eventhubscommon/client/EventHubsClientWrapper.scala
+++ b/core/src/main/scala/org/apache/spark/eventhubscommon/client/EventHubsClientWrapper.scala
@@ -22,7 +22,7 @@ import scala.collection.JavaConverters._
 
 import EventHubsOffsetTypes.EventHubsOffsetType
 import com.microsoft.azure.eventhubs.{EventHubClient => AzureEventHubClient, _}
-import com.microsoft.azure.servicebus._
+import com.microsoft.azure.eventhubs._
 
 import org.apache.spark.{SparkEnv, TaskContext}
 import org.apache.spark.eventhubscommon.EventHubNameAndPartition
diff --git a/core/src/main/scala/org/apache/spark/eventhubscommon/client/RestfulEventHubClient.scala b/core/src/main/scala/org/apache/spark/eventhubscommon/client/RestfulEventHubClient.scala
index 56676bd1..8fcfe223 100644
--- a/core/src/main/scala/org/apache/spark/eventhubscommon/client/RestfulEventHubClient.scala
+++ b/core/src/main/scala/org/apache/spark/eventhubscommon/client/RestfulEventHubClient.scala
@@ -26,7 +26,7 @@ import scala.concurrent.duration._
 import scala.util.{Failure, Success}
 import scala.xml.XML
 
-import com.microsoft.azure.servicebus.SharedAccessSignatureTokenProvider
+import com.microsoft.azure.eventhubs.SharedAccessSignatureTokenProvider
 import scalaj.http.{Http, HttpResponse}
 
 import org.apache.spark.eventhubscommon.EventHubNameAndPartition