Fix so you can close success screen