fix unchecked type issue