fixes bug from last commit