Fix Edit tests