Bit of a hack, but I'm OK with this.