Fixed update test to use find method