diff --git a/.travis.yml b/.travis.yml
index 9dbb61d..0aeab50 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,7 +1,7 @@
 language: python
 python:
   - "2.7"
-  
+
 before_install:
     - pip install pytest pytest-cov
 
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/logs/__init__.py b/tests/logs/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/logs/test_log_view.py b/tests/logs/test_log_view.py
index 94ade0f..d43a636 100644
--- a/tests/logs/test_log_view.py
+++ b/tests/logs/test_log_view.py
@@ -1,19 +1,15 @@
-import unittest
 
 from modules.logs import LogView
 
 
-class TestLogView(unittest.TestCase):
+class TestLogView(object):
 
     def test_get_all_logfiles(self):
+
         LogView._log_directory = "./logs"
 
         log_view = LogView()
         logfiles = log_view.get_all_logfiles()
 
         print(logfiles)
-        self.assertTrue(len(logfiles) > 0)
-
-
-if __name__ == '__main__':
-    unittest.main()
+        assert (len(logfiles) == 2)