diff --git a/models/__init__.py b/models/__init__.py
index 3173aa0..77ea3f3 100644
--- a/models/__init__.py
+++ b/models/__init__.py
@@ -1,5 +1,4 @@
-from .Pointnet2SemSeg import Pointnet2MSG as Pointnet2SemMSG
-from .Pointnet2SemSeg import Pointnet2SSG as Pointnet2SemSSG
-
-from .Pointnet2Cls import Pointnet2MSG as Pointnet2ClsMSG
-from .Pointnet2Cls import Pointnet2SSG as Pointnet2ClsSSG
+from .pointnet2_msg_sem import Pointnet2MSG as Pointnet2SemMSG
+from .pointnet2_ssg_sem import Pointnet2SSG as Pointnet2SemSSG
+from .pointnet2_msg_cls import Pointnet2MSG as Pointnet2ClsMSG
+from .pointnet2_ssg_cls import Pointnet2SSG as Pointnet2ClsSSG
diff --git a/train_cls.py b/train_cls.py
index b0594c5..b998e58 100644
--- a/train_cls.py
+++ b/train_cls.py
@@ -10,7 +10,7 @@ import os
 import tensorboard_logger as tb_log
 
 from models import Pointnet2ClsMSG as Pointnet
-from models.Pointnet2Cls import model_fn_decorator
+from models.pointnet2_msg_cls import model_fn_decorator
 from data import ModelNet40Cls
 import utils.pytorch_utils as pt_utils
 import data.data_utils as d_utils
diff --git a/train_sem_seg.py b/train_sem_seg.py
index 995e9d0..d41b343 100644
--- a/train_sem_seg.py
+++ b/train_sem_seg.py
@@ -9,8 +9,8 @@ import numpy as np
 import tensorboard_logger as tb_log
 import os
 
-from models import Pointnet2SSG as Pointnet
-from models.Pointnet2SemSeg import model_fn_decorator
+from models import Pointnet2SemMSG as Pointnet
+from models.pointnet2_msg_sem import model_fn_decorator
 from data import Indoor3DSemSeg
 import utils.pytorch_utils as pt_utils