fixes for ml