diff --git a/docker-compose.yml b/docker-compose.yml
index 33005c5..299474d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -25,7 +25,7 @@ services:
     shm_size: '4g'
     # ports:
     #   - 8000:8000
-    # command: uvicorn --host 0.0.0.0 src.api:app --reload
+    command: uvicorn --host 0.0.0.0 src.api:app --reload
     networks:
       - nginx
 
diff --git a/src/predict.py b/src/predict.py
index ea2a829..e9f890a 100644
--- a/src/predict.py
+++ b/src/predict.py
@@ -72,7 +72,7 @@ def compute_drug_embedding(
         # # In this case we vectorize one by one, so only 1 row in the array
         # embeddings = vectors[0].tolist()
         # TODO: add label also?
-        embeddings = list(embed_dict[drug_smiles])
+        embeddings: np.array = embed_dict[drug_smiles].tolist()
         vector_list.append({
             "vector": embeddings,
             "payload": {