Wip/bugfix