function fix :smile: