Modify test for getById