Adding missing header include