Could refactor it, however the logic is so small and almost non redudant. It's a bit more verbose, but easier to read IMHO ;) 
Took me a while to get your `for` loop construction, never used or saw it that way before.