1 files changed, 4 insertions, 8 deletions
diff --git a/cli/clean.py b/cli/clean.py
index b23a80e..90fa2b5 100644
--- a/cli/clean.py
+++ b/cli/clean.py
@@ -1,13 +1,9 @@
-import ymlstash
-
-from .base import cli, NAMES_DIR
-from .model import Name
+from .base import cli, STASH
 
 
 @cli.command()
 def clean():
-    stash = ymlstash.YmlStash(Name, NAMES_DIR, filter_none=True)
-    names = stash.list_keys()
+    names = STASH.list_keys()
     for name in names:
-        obj = stash.load(name)
-        stash.save(obj)
+        obj = STASH.load(name)
+        STASH.save(obj)