diff --git a/dynamight/__init__.py b/dynamight/__init__.py
index 32b274d..f5e133c 100644
--- a/dynamight/__init__.py
+++ b/dynamight/__init__.py
@@ -1,5 +1,5 @@
 from ._cli import cli
 from .inverse_deformations import optimize_inverse_deformations
-from .deformations import optimize_deformations
+from .deformations import optimize_deformations, optimize_deformations_rigid
 from .deformable_backprojection import deformable_backprojection
 from .evaluation import explore_latent_space