Filter robust to train test differences