small change to contract migration