add entity height back