Update tests to include type_checking.