Fix failing test.

@stephenminded @christhomson 

Please review. kthnxbye.