fix some methods not to refer removed type