Add type in _install function