Bug fixes 1 / 4