Update tests

@montskak so embarassed