Minor unit test fix.