Finally got a test working properly