Fixed create/update bug