Restore filter for test_remove rather than generator expression