oops

forgot a couple lines