Pushed a build fix for the test break