fix some re-usability issues with logit