[IMP] Correction from review