bug fixes

- Jonas