small fixes
2.0.2