fixed the feature i wanted to have in the last commit