Update unit.js

Fix tests