Corrected mixup

Corrected mixup