ML model working again