Add test for desired behaviour