Fixed grayscale undo
Small fixes