[IMP] Bug fix