refactor(test): typing cleanups