diff --git a/dvc/command/data_sync.py b/dvc/command/data_sync.py
index 7c01c8aaf..1bd28eb8c 100644
--- a/dvc/command/data_sync.py
+++ b/dvc/command/data_sync.py
@@ -52,7 +52,7 @@ class CmdDataStatus(CmdBase):
         with DvcLock(self.is_locker, self.git):
             status = self.cloud.status(self.parsed_args.targets, self.parsed_args.jobs)
 
-            if len(status) != len(self.parsed_args.targets):
+            if len(list(status)) != len(self.parsed_args.targets):
                 return 1
 
             self._show(status)
diff --git a/dvc/main.py b/dvc/main.py
index 9a7de66dc..11de00049 100644
--- a/dvc/main.py
+++ b/dvc/main.py
@@ -4,6 +4,7 @@ main entry point / argument parsing for dvc
 
 import sys
 from dvc.settings import Settings
+from dvc.logger import Logger
 
 def main():
     try: