fix name logic when function is used
0.2.2