another small fix to contract