Fix failed unittest