Fixed up failing build