Fix broken compile