Your code looks correct to do what you describe. In what way doesn’t it work?
What template is running the loop?
Are you sure the filter code is running? You could try placing a var_dump inside the filter inside the second if check and see if it shows on the page.
