Fixes test broken in 9909b5e
Fixes tests broken in 5374c0901ccc