Contract fix :)