Add test case for the changed behaviour