Fixed error when accessing User model. Apparently this was included in the upgrade docs