Correct a bug created by the previous commit