diff --git a/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py b/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py
index 3fb493537e..ac6fd3f574 100644
--- a/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py
+++ b/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py
@@ -14,5 +14,6 @@
 # limitations under the License.
 
 from .deepfm import DeepFM
+from .afm import AttentionalFM  
 
-__all__ = ["DeepFM"]
+__all__ = ['DeepFM', 'AttentionalFM']