ai fixes

light fixes by Filip