Fix gen 2 crash