Add missing header include