Good point - I had actually tried to remove the class from the wrong element. Fixed it now and added a test to ensure it works. Please take another look if you have a free minute.

Thanks