Minor change to be more idiomatic