Undo bug fixed

Can now successfully undo any play without bugs