Fix to match the output