Fixed minor extra code.