Changed test according to feedback from Liam Mullane