bug fixes 0.5.4