Fix unittest failure