Fix test type checking