(Minor) Fixed: missing header file in previous commit