fixing error after change