Quick fixes to get ai-dev running