You can make a one-liner of this and get rid of the previous assignment