unittest error fix