diff --git a/genevector/embedding.py b/genevector/embedding.py
index 9a19e44..fef5c71 100644
--- a/genevector/embedding.py
+++ b/genevector/embedding.py
@@ -562,7 +562,7 @@ def get_inverse_predictive_genes(self, adata, label, n_genes=10):
             vectors[x] = vector
         markers = dict()
         for x, mvec in vectors.items():
-            ct_sig = self.embed.get_similar_genes(mvec)[-1.*n_genes:]["Gene"].tolist()
+            ct_sig = self.embed.get_similar_genes(mvec)[int(-1.*n_genes):]["Gene"].tolist()
             markers[x] = ct_sig
         return markers
 
@@ -739,7 +739,7 @@ def cluster(self, adata, up_markers, down_markers=dict()):
         down_markers = dict()
         down_only = set(down_markers.keys()).difference(set(up_markers.keys()))
         up_and_down = set(down_markers.keys()).intersection(set(up_markers.keys()))
-        phs = set(up_markers.keys()).union(set(down_markers.keys()))
+        phs = list(set(up_markers.keys()).union(set(down_markers.keys())))
         for ph in up_only:
             genes = up_markers[ph]
             vec = self.embed.generate_vector(genes)