Bug fixed for final presentation

Good luck！