fixed ai bugs