fix the failing unittest