Resolved #22, fix a mistake