minor fix to pass test fail