diff --git a/python/flex_joints/__init__.py b/python/flex_joints/__init__.py
index e69de29..53bb913 100644
--- a/python/flex_joints/__init__.py
+++ b/python/flex_joints/__init__.py
@@ -0,0 +1 @@
+from .flex_joints import Flex  # noqa: F401
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index f72c3ad..d3352f8 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -1,3 +1,7 @@
 ADD_UNIT_TEST(flex-estimator test-flex-estimator.cpp)
 target_link_libraries(flex-estimator PUBLIC ${PROJECT_NAME})
 target_link_libraries(flex-estimator PRIVATE Boost::unit_test_framework)
+
+if(BUILD_PYTHON_INTERFACE)
+  ADD_PYTHON_UNIT_TEST("py-flexi-hips" "tests/python/test_flexi_hips.py" "python")
+endif()