Fixed an issue with issue 7 unit test