Fix the progress bar issue