fixed weird issue in unit tests