diff --git a/fit_single_img.py b/fit_single_img.py
index 5fefa76..4825da1 100644
--- a/fit_single_img.py
+++ b/fit_single_img.py
@@ -16,7 +16,7 @@ def fit(args):
     print('loading models')
     mtcnn = MTCNN(device=args.device, select_largest=False)
     fa = face_alignment.FaceAlignment(
-        face_alignment.LandmarksType._3D, flip_input=False)
+        face_alignment.LandmarksType.THREE_D, flip_input=False)
     recon_model = get_recon_model(model=args.recon_model,
                                   device=args.device,
                                   batch_size=1,
diff --git a/fit_video.py b/fit_video.py
index 34b8987..dfe394a 100644
--- a/fit_video.py
+++ b/fit_video.py
@@ -277,7 +277,7 @@ def process_video(args, device):
     mtcnn = MTCNN(device=device, select_largest=False)
 
     fa = face_alignment.FaceAlignment(
-        face_alignment.LandmarksType._3D, flip_input=True, device=device)
+        face_alignment.LandmarksType.THREE_D, flip_input=True, device=device)
 
     frame_ind = 0