Minor bugfixes...