Fixed broken build (by including missing headers).