Fixing small bug with header