Correct bug for multi modal