quick fix of the bug
fixing tests