Oops, compile errors

Fixed minor compile errors from last commit