Fix issues noted in unit tests in https://github.com/astronomer/orbiter/pull/83/files#diff-4c2e95c65a465b4b8cf7eea2566f730a85659881e41afe7675f42121e0cac795R20-R48