Make it work on search pages