Fix gen update bug