DOC: simplify the code (Thx Stefan)