should fix #99 but will leave open for more testing

also fixes error if user undoes everything