Fixed an issue with type hinting in unit tests.