Fixing a failing test