fix the ML problem, now ML is functioning