Very hacky. There must be a better way of doing this.