FIX: minor fix on the draft