Should fix build error