Test fix

@bpostlethwaite for reference