Fix include bug