This should fix the mistakes in the test.