Fix for AI check (DZAI/etc)

Provided problems with debugging, not needed until ZAI release.