fix small issure