Add test for unit typechecking