TODO: Rewrite this.