Fix scheduler build