diff --git a/bin/2.2_motif_estimation/get_best_motif.py b/bin/2.2_motif_estimation/get_best_motif.py
index beacb1c..7f9589d 100644
--- a/bin/2.2_motif_estimation/get_best_motif.py
+++ b/bin/2.2_motif_estimation/get_best_motif.py
@@ -7,6 +7,7 @@ def parse_arguments():
     parser.add_argument("meme", help="Path to 'meme' file generated by GLAM2")
     parser.add_argument("output", help="Output file")
     parser.add_argument("num", help="Number of motifs parsed from file")
+    parser.add_argument("id", help="Cluster ID")
     args = parser.parse_args()
     return args
 
@@ -36,7 +37,7 @@ def main():
     break_header = "MOTIF  " + str(number)
 
     # Get cluster_id
-    cluster_id = os.path.splitext(args.meme)[0].split('/')[-2].split('_')[-1]
+    cluster_id = args.id
 
     # Pattern for motif header
     pattern = re.compile("^MOTIF\s{2}(\d)+")