Bug fixes 2 / 5