#786 Fixed unit test