Fix missing cancel button