diff --git a/django_diu/import_command.py b/django_diu/import_command.py
index b971e3c50fab0b79eb5e3b3ec3f46dc3eaf9d944..75932fa65ac1df8ffed963518d34d4ddd54de55b 100644
--- a/django_diu/import_command.py
+++ b/django_diu/import_command.py
@@ -152,6 +152,9 @@ class DataFrameImportTask(ImportTask):
     def open_data_source(self):
         self.rows = self.dataframe.iterrows()
 
+    def count_source(self):
+        self.source_count = self.dataframe.shape[0]
+
 
 class ImportCommand(BaseCommand):
     help = "Generic command to import data into a django database"