small fix 2.0