Fixed test case, perhaps there is a better way to do this.