quick fix, do not know what is better way to do it