fix a couple lines