fixed bug from last commit, tests pass

Sorry about this.