fix unit test regression