Fix the header file

Forgot this one