fix build after refactory