update unit tests

\*sigh\*