diff --git a/docs/index.rst b/docs/index.rst
index 946234cd4d0d55f896d9eb8792adf1dfca0b5e91..b940ae7bbe97eefeeaed64dd81ff378708778ae6 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -84,7 +84,7 @@ You can also use the class :class:`pystencils_autodiff.AutoDiffOp` to obtain bot
     op = AutoDiffOp(forward_assignments)
     backward_assignments = op.backward_assignments   
 
-    tensorflow_op = op.create_tensorflow_op(backend='tensorflow_native')
+    tensorflow_op = op.create_tensorflow_op(backend='tensorflow_native', use_cuda=False)
 
 .. testoutput::
    :hide:
@@ -97,7 +97,7 @@ You can also use the class :class:`pystencils_autodiff.AutoDiffOp` to obtain bot
 
 .. testcode::
 
-    torch_op = op.create_tensorflow_op(backend='torch_native')
+    torch_op = op.create_tensorflow_op(backend='torch_native', use_cuda=False)
 
 
 Contents