Small build fix