Bug fix in test code
documentation