Fixed the test to better reflect what's happening