Fix breaking contract update