fix a couple of things that were missed