Change return type for Create and Update hooks