diff --git a/term_frag_sel/fragmentation.py b/term_frag_sel/fragmentation.py
index f3471a0d96d5b80996639f1185bfde8abfe769ba..9cd8a0ff760a81bc17cb6673ec80dcb932a0d290 100644
--- a/term_frag_sel/fragmentation.py
+++ b/term_frag_sel/fragmentation.py
@@ -24,7 +24,7 @@ def fragmentation(fasta: dict, seq_counts: pd.DataFrame,
 
     term_frags = []
     for seq_id, seq in fasta.items():
-        counts = seq_counts[seq_counts["seqID"] == str(seq_id)]["count"]
+        counts = seq_counts[seq_counts["seqID"] == seq_id]["count"]
         for _ in range(counts.iloc[0]):
             cuts = []
             seq_len = len(seq)