Couple bugfixes