Small update to contract