Minor fix to restart build