fixed the enhancement issue