Main: added small fix