#786 Fixed broken unit test