diff --git a/pytato/__init__.py b/pytato/__init__.py
index 4a89b01bb775e83fda9a49fce886643c01afb624..6f4d18acb27f53be1a05fdb2ed1629d69622eb05 100644
--- a/pytato/__init__.py
+++ b/pytato/__init__.py
@@ -26,7 +26,7 @@ THE SOFTWARE.
 
 from pytato.array import (
         Array, AbstractResultWithNamedArrays, DictOfNamedArrays, Placeholder,
-        IndexLambda, NamedArray,
+        IndexLambda, NamedArray, DataWrapper, InputArgumentBase,
 
         make_dict_of_named_arrays,
         make_placeholder, make_size_param, make_data_wrapper,
@@ -62,6 +62,7 @@ from pytato.visualization import get_dot_graph, show_dot_graph
 __all__ = (
         "Array", "AbstractResultWithNamedArrays", "DictOfNamedArrays",
         "Placeholder", "IndexLambda", "NamedArray", "LoopyCall",
+        "DataWrapper", "InputArgumentBase",
 
         "make_dict_of_named_arrays", "make_placeholder", "make_size_param",
         "make_data_wrapper", "einsum",