Fix for mongo update