fix the type checking code
2 files changed