add extra test and fix it