Small change to unit tests