Fixed the error when runing unit test