Fix small issue with test which was causing it to fail