fix a couple things