Some AI cleanup

just a tiny commit with the things I forgot 8)