Forgot to update type check.