Add type check