fix build error from before