fix tests

@anthonybishopric