bugfix + unittest