fix build with after update