Fix/minor bugfixes