diff --git a/.travis.yml b/.travis.yml
index 10973e3..110fb48 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -5,6 +5,8 @@ python:
   - "3.5"
   - "3.6"
 script:
+  - pycodestyle django_migration_linter
+  - flake8 django_migration_linter
   - py.test
 install:
   - pip install -e ".[test]"
diff --git a/django_migration_linter/migration_linter.py b/django_migration_linter/migration_linter.py
index 40f388f..38de743 100644
--- a/django_migration_linter/migration_linter.py
+++ b/django_migration_linter/migration_linter.py
@@ -109,7 +109,8 @@ class MigrationLinter:
         diff_process.wait()
         if diff_process.returncode != 0:
             output = []
-            for line in map(clean_bytes_to_str, diff_process.stderr.readlines()):
+            for line in map(
+                    clean_bytes_to_str, diff_process.stderr.readlines()):
                 output.append(line)
             log.info("Error while git diff command:\n{}".format(
                 "".join(output)))
@@ -187,7 +188,8 @@ class MigrationLinter:
         sqlmigrate_process = Popen(
             git_diff_command, shell=True, stdout=PIPE, stderr=PIPE)
         sql_statements = []
-        for line in map(clean_bytes_to_str, sqlmigrate_process.stdout.readlines()):
+        for line in map(
+                clean_bytes_to_str, sqlmigrate_process.stdout.readlines()):
             if not line.startswith('--'):  # Ignore comments
                 sql_statements.append(line)
         sqlmigrate_process.wait()
diff --git a/setup.py b/setup.py
index 7fa637c..f5a7f85 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,9 @@ install_requirements = [
 test_requirements = [
     'pytest',
     'django',
-    'django-fake-database-backends'
+    'django-fake-database-backends',
+    'pycodestyle',
+    'flake8'
 ]
 
 setup(