small bugfixed, adding missing includes.