diff --git a/components/AllTrainDiagramView.js b/components/AllTrainDiagramView.js
index 89801bc..a92ca21 100644
--- a/components/AllTrainDiagramView.js
+++ b/components/AllTrainDiagramView.js
@@ -174,9 +174,8 @@ export default function AllTrainDiagramView() {
               return false;
             }
           }
-          const { img, trainName, type, trainNumDistance, infogram } =
-            customTrainDataDetector(d, allCustomTrainData);
-          return d.includes(input) || trainName.includes(input);
+          const { img, trainName, type, trainNumDistance, infogram, TrainNumberOverride } = customTrainDataDetector(d, allCustomTrainData);
+          return d.includes(input) || trainName.includes(input) || (TrainNumberOverride && TrainNumberOverride.includes(input));
         })}
         renderItem={({ item }) => <Item {...{ openTrainInfo, id: item }} />}
         ListEmptyComponent={