Fixed error caused by missing header