Fix unit tests from copy/paste error