small revision for the test to pass