minor fix
minor fixes 2.0