fix tests

@dcadams