Refac: Fixed ast type for visitor argument