fix header for demo code