Fix test for new id system