Minor fix to addressed failed test