Minor: missed part from prevoius commit