Fix getting wrong id on test