Bugfixes, tests