update the related code since the header issue is fixed