fix compilation

forgot a couple things