fix unit testing for ci