Fix it breaking when you press Cancel