Stepped shaft with keyway