small fixes [0.2.1]