Add include to get rid of last compilation errors