Update tests

there :D