refactor: scaled to correct size in AI