diff --git a/docs/notebooks/recovering_from_errors.pct.py b/docs/notebooks/recovering_from_errors.pct.py
index a3116bee2f..a4de11a58b 100644
--- a/docs/notebooks/recovering_from_errors.pct.py
+++ b/docs/notebooks/recovering_from_errors.pct.py
@@ -1,9 +1,9 @@
 # %% [markdown]
 # # Recovering from errors
 
+# %%
 import random
 
-# %%
 import numpy as np
 import tensorflow as tf