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

ok, that was michal's idea :p