diff --git a/static_sitemaps/management/commands/refresh_sitemap.py b/static_sitemaps/management/commands/refresh_sitemap.py
index f5452cf..2f3654b 100644
--- a/static_sitemaps/management/commands/refresh_sitemap.py
+++ b/static_sitemaps/management/commands/refresh_sitemap.py
@@ -1,14 +1,13 @@
-from django.core.management.base import NoArgsCommand
-
+from django.core.management.base import BaseCommand
 from static_sitemaps.generator import SitemapGenerator
 
 __author__ = 'xaralis'
 
 
-class Command(NoArgsCommand):
+class Command(BaseCommand):
     command = None
     help = 'Generates sitemaps files to a predefined directory.'
 
-    def handle_noargs(self, **options):
+    def handle(self, *args, **options):
         generator = SitemapGenerator(int(options.get('verbosity', 0)))
         generator.write()