Fix unittest dependency