add `Unit` support (good for testing)