Need help to fix a failing test