Update test to not rely on clean DOM