Fix missplaced unit test