diff --git a/bigchaindb/backend/localmongodb/query.py b/bigchaindb/backend/localmongodb/query.py index aea34bd9..d193ebb9 100644 --- a/bigchaindb/backend/localmongodb/query.py +++ b/bigchaindb/backend/localmongodb/query.py @@ -322,14 +322,12 @@ def get_validator_set(conn, height=None): def get_election(conn, election_id): query = {'election_id': election_id} - cursor = conn.run( + return conn.run( conn.collection('elections') - .find(query, projection={'_id': False}) - .sort([('height', DESCENDING)]) + .find_one(query, projection={'_id': False}, + sort=[('height', DESCENDING)]) ) - return next(cursor, None) - @register_query(LocalMongoDBConnection) def get_asset_tokens_for_public_key(conn, asset_id, public_key):