Fix build fix