Fix entity remove