Fix entity bugs