diff --git a/genomap/utils/util_Sig.py b/genomap/utils/util_Sig.py
index e8c143c..98fe286 100644
--- a/genomap/utils/util_Sig.py
+++ b/genomap/utils/util_Sig.py
@@ -56,6 +56,8 @@ def createGenomap_for_sig(data,gene_names,rowNum=32,colNum=32):
     if nump<data.shape[1]:
         data,index=select_n_features(data,nump)
         gene_namesRe=gene_names[index]
+    else:
+        gene_namesRe=gene_names
       #  X_train=X_train[top_n_features]
     #ldaOutzc=scipy.stats.zscore(data, axis=0, ddof=1)    
     genoMaps,T=construct_genomap_returnT(data,rowNum,colNum,epsilon=0.0,num_iter=200)
diff --git a/setup.py b/setup.py
index 477fe1e..c4998da 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="genomap",
-    version="1.3.5",
+    version="1.3.6",
     author="Md Tauhidul Islam",
     author_email="tauhid@stanford.edu",
     description="Genomap converts tabular gene expression data into spatially meaningful images.",