Update test to work properly on IE