Small update to type check