Continue fixing the function.