Failing test minor fix