Slight fix in unit tests