[IMP] improve type