Change the test accordingly