Fix Linux compilation

Fix some includes so that atom compiles successfully on Linux.