I need to test before I push...

* Fixed error in `unit` function