couple bugfixes