AI quickfix

hopefully works