diff --git a/P1. Baseline.ipynb b/P1. Baseline.ipynb
index 58d6b8a..793ceb5 100644
--- a/P1. Baseline.ipynb	
+++ b/P1. Baseline.ipynb	
@@ -809,7 +809,7 @@
     "        \n",
     "        max_row_mean=np.max(row_means)\n",
     "        row_means[row_means==0]=max_row_mean+1\n",
-    "        to_subtract_rows=sparse.diags(row_means)*(result>0)\n",
+    "        to_subtract_rows=sparse.diags(row_means)*(result.power(0))\n",
     "        to_subtract_rows.sort_indices() # needed to have valid .data\n",
     "        \n",
     "        subtract=to_subtract_rows.data\n",
@@ -1245,7 +1245,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.9"
+   "version": "3.8.5"
   }
  },
  "nbformat": 4,