AI
 - Fix up minor issue.