diff --git a/src/pystencils_autodiff/framework_integration/printer.py b/src/pystencils_autodiff/framework_integration/printer.py
index 922535228d705ca4088bd36ae0bd26dc7f13025b..40e8e0b8e6d2cad2f94a70502a4385d6e85a4f3b 100644
--- a/src/pystencils_autodiff/framework_integration/printer.py
+++ b/src/pystencils_autodiff/framework_integration/printer.py
@@ -195,3 +195,4 @@ class DebugFrameworkPrinter(FrameworkIntegrationPrinter):
 
 
 show_code = functools.partial(pystencils.show_code, custom_backend=FrameworkIntegrationPrinter())
+get_code_str = functools.partial(pystencils.get_code_str, custom_backend=FrameworkIntegrationPrinter())