I guess my original comment was mainly about removing the unnecessary `.apply`, but I'll take it as-is.