fix 2 bugts