[tests]: Fixed tests that had broken