diff --git a/python/src/main.cpp b/python/src/main.cpp
index 5fe74f021..e3b7fb371 100644
--- a/python/src/main.cpp
+++ b/python/src/main.cpp
@@ -21,7 +21,10 @@ PYBIND11_MODULE(komputepy, m) {
         .def(py::init(
             [](const std::vector<float>& data, kp::Tensor::TensorTypes tensorTypes) {
                 return std::unique_ptr<kp::Tensor>(new kp::Tensor(data, tensorTypes));
-            }));
+            }))
+        .def("data", &kp::Tensor::data);
+
+    py::class_<kp::OpBase>(m, "OpBase");
 
 #ifdef VERSION_INFO
     m.attr("__version__") = VERSION_INFO;