Fixed the failing unit test 

introduced by 35191226c9ef88990992700dfed3c32b78edf78f