added missng use
smal lrefactor