Bug fixes 2 / 4