Fixed the test for good measure