Tried tweaking the test