Fixed minor include error