Small update to the test