Minor bugfixes ;)