Sorry for not solving it more elegantly before.