Fixed Model Admin Draft Issues

Now works with 3.1.1+
Thanks @icecaster, silverstripe-versioned-gridfield made this fix much
easier!