fixing bad ending