Make smal fix