Fixed minor breakage from previous commit