diff --git a/python/mujoco/usd/demo.py b/python/mujoco/usd/demo.py
index 645fa2e4..069589e4 100644
--- a/python/mujoco/usd/demo.py
+++ b/python/mujoco/usd/demo.py
@@ -3,8 +3,7 @@ from tqdm import tqdm
 from pathlib import Path
 
 import mujoco
-# from mujoco.usd import exporter
-import exporter as exporter
+from mujoco.usd import exporter
 
 def generate_usd_trajectory(args):
 
diff --git a/python/mujoco/usd/exporter.py b/python/mujoco/usd/exporter.py
index 371ca5bf..89d9fca0 100644
--- a/python/mujoco/usd/exporter.py
+++ b/python/mujoco/usd/exporter.py
@@ -16,12 +16,10 @@ import os
 
 import mujoco
 
-# import mujoco.usd.shapes as shapes_module
-# import mujoco.usd.component as object_module
-import shapes as shapes_module
-import objects as object_module
-import lights as light_module
-import camera as camera_module
+import mujoco.usd.shapes as shapes_module
+import mujoco.usd.objects as object_module
+import mujoco.usd.lights as light_module
+import mujoco.usd.camera as camera_module
 
 import numpy as np
 import scipy
diff --git a/python/mujoco/usd/objects.py b/python/mujoco/usd/objects.py
index af710351..4bf32cc1 100644
--- a/python/mujoco/usd/objects.py
+++ b/python/mujoco/usd/objects.py
@@ -19,10 +19,8 @@ import pprint
 
 import mujoco
 
-# import mujoco.usd.utils as utils_component
-# import mujoco.usd.shapes as shapes_component
-import utils as utils_component
-import shapes as shapes_component
+import mujoco.usd.utils as utils_component
+import mujoco.usd.shapes as shapes_component
 
 import numpy as np