Test passes with the above change.