Correct the tests to validate expected behaviour