Corrent bad return value and add missing #include