Switched from model name to type