[#369] Fixed broken ml test that sam didnt tell me about
[#369] Fixed broken ml test that sam didnt tell me about

Cyberjusticelab/JusticeAI,Cyberjusticelab/JusticeAI,Cyberjusticelab/JusticeAI,Cyberjusticelab/JusticeAI