Update tests using @test macro