Updates tests
Updates tests