Fix gen 2 bug