diff --git a/tests/trainers/test_ofa_trainer.py b/tests/trainers/test_ofa_trainer.py
index 46dc5c8b..75b8cbbf 100644
--- a/tests/trainers/test_ofa_trainer.py
+++ b/tests/trainers/test_ofa_trainer.py
@@ -98,7 +98,7 @@ class TestOfaTrainer(unittest.TestCase):
 
         self.assertIn(ModelFile.TORCH_MODEL_BIN_FILE,
                       os.listdir(os.path.join(WORKSPACE, 'output')))
-        # shutil.rmtree(WORKSPACE)
+        shutil.rmtree(WORKSPACE)
 
 
 if __name__ == '__main__':