minor fix after 2.1.0