some fixes for mi classifier