Fixed Unittest fail on mock object