Your first mockup there is what I assumed they were going to do.

Then they did this.

Your's is better.