Need a better solution to this.