Fix build

First build fix.