Fix build

Second build fix :).