Fix entity update callback not being called which was broken in previous commit.