fix unit tests
1 file changed