Change test according to the new behavior