Fix includes so code can compile

I fixed includes so the code can now compile.