Functions that are decorated with `trace` are now properly typed and the type hints for them are fixed.