Fixed little issue