fix build

quick fix