Fix unittest/