diff --git a/.travis.yml b/.travis.yml
index f7a9264803f8..be227df1fdbd 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -6,7 +6,7 @@ before_install: pip install --upgrade pip setuptools
 install: pip install -r requirements.txt
 before_script:
   - black --check . || true
-  - flake8 . --count --select=E9,F401,F63,F7,F82 --show-source --statistics
+  - flake8 . --count --select=E9,F4,F63,F7,F82 --show-source --statistics
 script:
   - scripts/validate_filenames.py  # no uppercase, no spaces, in a directory
   - mypy --ignore-missing-imports .
diff --git a/linear_algebra/src/tests.py b/linear_algebra/src/tests.py
index a26eb92653e2..afca4ce87117 100644
--- a/linear_algebra/src/tests.py
+++ b/linear_algebra/src/tests.py
@@ -9,7 +9,7 @@
 """
 
 import unittest
-from lib import *
+from lib import Matrix, Vector, axpy, squareZeroMatrix, unitBasisVector, zeroVector
 
 class Test(unittest.TestCase):
     def test_component(self):