diff --git a/.flake8 b/.flake8
index ebf130ec..fa836f63 100644
--- a/.flake8
+++ b/.flake8
@@ -4,7 +4,7 @@ select = A,B,B9,BLK,C,E,F,I,N,S,W
 ignore = E203,W503,E501
 max-complexity = 10
 max-line-length = 80
-application-import-names = common,src,test,util
+application-import-names = datagateway_api,test,util
 import-order-style = google
 per-file-ignores = test/*:S101
 enable-extensions=G