Fixed an error after removing type