passing `Unit` as a `unit` kwarg now works as expected