diff --git a/vectorlm/utils/__init__.py b/vectorlm/utils/__init__.py
index ba14dac..0190f5d 100644
--- a/vectorlm/utils/__init__.py
+++ b/vectorlm/utils/__init__.py
@@ -12,12 +12,10 @@
     checkpoint_exists,
     get_latest_checkpoint_dir,
     load_metadata,
-    load_model,
-    load_optimizer,
+    load_model_and_optimizer,
     load_scheduler,
     save_consolidated_model,
     save_metadata,
-    save_model,
-    save_optimizer,
+    save_model_and_optimizer,
     save_scheduler,
 )