diff --git a/test_data/test_classification.py b/test_data/test_classification.py
index b5ef11cd69454f92b4fb167eaa37002eeb6f2490..711eafdef9a608cfb3328abb74c7f50b7dcacd3b 100644
--- a/test_data/test_classification.py
+++ b/test_data/test_classification.py
@@ -10,7 +10,7 @@ def main():
     truths = []
     qc_percents = []
     for problem_file in glob('AE*/problems.csv'):
-        for qc_file in glob(os.path.join(os.path.dirname(problem_file), '*.qc')):
+        for qc_file in glob(os.path.join(os.path.dirname(problem_file), '*QC.nc')):
             qc = netCDF4.Dataset(qc_file)
             break
         else: