Ah my bad, wasn't following the logic closely enough.