Fix build due to removal