Added missing header include