fix to pass unittest