Bugfixes I hope...