Small AI Fixes