fixes bug in last commit