Update tests :wrench: