Fix test program compilation: Add missing #include and update the call