Change headers to fix build error on Windows

Change headers to fix build error on Windows