Forgot to add updated header file