Tests build fix