Add `herokaiOnly`