Updated my code as I wasn't right. Turns out it's much more awkward :(