Try to fix build fail at linux

There were missing some includes of .h files.