try to fix the unittest to be runable