diff --git a/pangolin/__init__.py b/pangolin/__init__.py
index 4822214..d8bd3b0 100644
--- a/pangolin/__init__.py
+++ b/pangolin/__init__.py
@@ -1,2 +1,2 @@
 _program = "pangolin"
-__version__ = "2.3.4"
+__version__ = "2.3.5"
diff --git a/pangolin/scripts/pangolearn.smk b/pangolin/scripts/pangolearn.smk
index e05c6d0..91c9710 100644
--- a/pangolin/scripts/pangolearn.smk
+++ b/pangolin/scripts/pangolearn.smk
@@ -295,7 +295,7 @@ rule overwrite:
                         new_row["lineage"] = "B.1.1.7"
 
                         writer.writerow(new_row)
-                    elif row["lineage"] =="B.1.351" and row["taxon"] not in b1351:
+                    elif row["lineage"].startswith("B.1.351") and row["taxon"] not in b1351:
                         new_row = row
                         
                         new_row["probability"] = "1.0"
@@ -303,7 +303,7 @@ rule overwrite:
 
                         writer.writerow(new_row)
                         
-                    elif row["taxon"] in b1351 and not row["lineage"].startswith("B.1.351."):
+                    elif row["taxon"] in b1351:
                         new_row = row
                         
                         snps = b1351[row["taxon"]]