Yes, that would have been more idiomatic. Feel free to update it if you like (in a new pull).