diff --git a/opencompass/datasets/needlebench/multi.py b/opencompass/datasets/needlebench/multi.py
index b100ad98..648d3a5b 100644
--- a/opencompass/datasets/needlebench/multi.py
+++ b/opencompass/datasets/needlebench/multi.py
@@ -1,4 +1,4 @@
-# noqa: E501
+# flake8: noqa: E501
 import json
 import os
 import random
diff --git a/opencompass/datasets/needlebench/origin.py b/opencompass/datasets/needlebench/origin.py
index 9bf7ba89..c976a3cc 100644
--- a/opencompass/datasets/needlebench/origin.py
+++ b/opencompass/datasets/needlebench/origin.py
@@ -1,4 +1,4 @@
-# noqa: E501
+# flake8: noqa: E501
 import json
 import os
 import random
diff --git a/opencompass/datasets/needlebench/parallel.py b/opencompass/datasets/needlebench/parallel.py
index c4eb3da9..87907401 100644
--- a/opencompass/datasets/needlebench/parallel.py
+++ b/opencompass/datasets/needlebench/parallel.py
@@ -1,4 +1,4 @@
-# noqa: E501
+# flake8: noqa: E501
 import json
 import os
 import random