Minor Bug fixed with Retry Button