fix unit tests (#328)