Maybe you should replace this with simple `use` & type-hints?