fix(unit): match @ units properly

Fixes #9.