Bugfixe in Unit test