Fixed issue with test