Fixed includes so that the build gem works!