Chore/bugfixes