diff --git a/TorrentToMedia.py b/TorrentToMedia.py
index 5536bbab..0e14e60d 100755
--- a/TorrentToMedia.py
+++ b/TorrentToMedia.py
@@ -2,7 +2,7 @@
 # coding=utf-8
 
 import cleanup
-cleanup.clean('core', 'libs')
+cleanup.clean(cleanup.FOLDER_STRUCTURE)
 
 import datetime
 import os
diff --git a/nzbToMedia.py b/nzbToMedia.py
index 2fc8c80c..fceed9cd 100755
--- a/nzbToMedia.py
+++ b/nzbToMedia.py
@@ -624,7 +624,7 @@
 from __future__ import print_function
 
 import cleanup
-cleanup.clean('core', 'libs')
+cleanup.clean(cleanup.FOLDER_STRUCTURE)
 
 import datetime
 import os