Fix small issue introduced in v2.0.1