Fix build

Fix build by using the right include directories.