diff --git a/python/sparknlp/embeddings.py b/python/sparknlp/embeddings.py
index <HASH>..<HASH> 100644
--- a/python/sparknlp/embeddings.py
+++ b/python/sparknlp/embeddings.py
@@ -7,6 +7,12 @@ import threading
 import time
 import sparknlp.pretrained as _pretrained
 
+
+# DONT REMOVE THIS IMPORT
+from sparknlp.annotator import WordEmbeddingsModel
+####
+
+
 class Embeddings:
     def __init__(self, embeddings):
         self.jembeddings = embeddings
[skip travis] import word embedding models as embeddings for jvm compatibility