Think it's a better solution here. I'll change like this.