Adapt unittest