Small compile fix