diff --git a/crisprbact/predict.py b/crisprbact/predict.py
index 0c2cd5b076710e21b0ef6d0bc44caf5f7dbc08c4..5b88c378304036f285b656717ed4ce7082e40ba4 100644
--- a/crisprbact/predict.py
+++ b/crisprbact/predict.py
@@ -43,7 +43,7 @@ def find_targets(seq):
         )
 
 
-def on_target_predict(seq, genome=None):
+def on_target_predict(seq, genome=None, seed_size=7):
 
     seq = seq.upper()  # make uppercase
     seq = re.sub(r"\s", "", seq)  # removes white space
@@ -63,11 +63,13 @@ def on_target_predict(seq, genome=None):
         )
         X = X.reshape(X.shape[0], -1)
         preds = predict(X)
+
         for i, target in enumerate(alltargets):
             target.update({"pred": preds[i]})
+            target.update({"seed_size": seed_size})
             if genome:
                 off_target_df = compute_off_target_df(
-                    target["guide"], 7, records, genome_features
+                    target["guide"], seed_size, records, genome_features
                 )
                 off_target_list = []
                 features = off_target_df.loc[0:, "features"]
diff --git a/pyproject.toml b/pyproject.toml
index 627a66f37b553141750abb14e76e8eedfaa4f3ff..12da22efc64bdadd0b7a0e7702de8129695efe4c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "crisprbact"
-version = "0.2.0"
+version = "0.2.1"
 license = "GPL-3.0"
 description = "Tools to design and analyse CRISPRi experiments"
 authors = ["David Bikard <david.bikard@pasteur.fr>", "Remi Planel <rplanel@pasteur.fr>"]