Fix build

@lennartcl