Improve cancel action on mod screen