Small compile error fix