Fix unit testing error.