diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py
index e15b7d2fd9ece8e4801e43fc85362a18684e4eb6..ea069114fbd29fad6c128c3c1cf46179fa0b1ddc 100644
--- a/src/pystencils_autodiff/__init__.py
+++ b/src/pystencils_autodiff/__init__.py
@@ -7,6 +7,7 @@ from pystencils_autodiff._adjoint_field import AdjointField
 from pystencils_autodiff._autodiff import (
     AutoDiffAstPair, AutoDiffBoundaryHandling, AutoDiffOp, DiffModes, create_backward_assignments,
     get_jacobian_of_assignments)
+from pystencils_autodiff.field_tensor_conversion import ArrayWrapper, coerce_to_field
 
 __all__ = ['backends',
            'AdjointField',
@@ -16,7 +17,9 @@ __all__ = ['backends',
            'AutoDiffAstPair',
            'tensorflow_jit',
            'DiffModes',
-           'AutoDiffBoundaryHandling']
+           'AutoDiffBoundaryHandling',
+           'ArrayWrapper',
+           'coerce_to_field']
 
 sys.modules['pystencils.autodiff'] = pystencils_autodiff
 sys.modules['pystencils.autodiff.backends'] = pystencils_autodiff.backends