Fixing the build

@maxfliri