Fix a failing testcase

This fix is a bit of a blind guess, I couldn't really discern
from recent commits whether this is an expected behavior change
or not.