Fix lại check extension