Failing test for second case, let's fix this.