bug fixes 0.1.2