Fixed issue with failing test