Fix header not matching implementation
Fix header not matching implementation