Update test case for mongoengine update