fix small ui thing