diff --git a/fast/diffusion_3D_wBCs.py b/fast/diffusion_3D_wBCs.py
index 9cb39d5179..9220e25247 100644
--- a/fast/diffusion_3D_wBCs.py
+++ b/fast/diffusion_3D_wBCs.py
@@ -76,7 +76,7 @@ def plot_3dfunc(u):
 x, y, z = grid.dimensions
 t = grid.stepping_dim
 
-print(eq_stencil)
+# print(eq_stencil)
 
 # Create Operator
 if args.devito:
diff --git a/fast/wave2d_b.py b/fast/wave2d_b.py
index f0fc4c13b5..ffc570578a 100644
--- a/fast/wave2d_b.py
+++ b/fast/wave2d_b.py
@@ -90,15 +90,14 @@
 # np.save("critical_dt%s.npy" % shape_str, model.critical_dt, allow_pickle=True)
 # np.save("wave_dat%s.npy" % shape_str, u.data[:], allow_pickle=True)
 
-if len(shape) == 2:
-    if args.plot:
-        plot_2dfunc(u)
+if len(shape) == 2 and args.plot:
+    plot_2dfunc(u)
 
 print("Init norm:", np.linalg.norm(u.data[:]))
 
 if args.devito:
     # Run more with no sources now (Not supported in xdsl)
-    #op1 = Operator([stencil], name='DevitoOperator', subs=grid.spacing_map)
+    # op1 = Operator([stencil], name='DevitoOperator', subs=grid.spacing_map)
     op1 = Operator([stencil], name='DevitoOperator')
     op1.apply(time=nt, dt=dt)
 
@@ -128,7 +127,7 @@
 
     # Run more with no sources now (Not supported in xdsl)
     xdslop = Operator([stencil], name='xDSLOperator')
-    xdslop.apply(time=nt)
+    xdslop.apply(time=nt, dt=dt)
 
     if len(shape) == 2 and args.plot:
         plot_2dfunc(u)