Fix include path
Fix possible buffer overflow