fix for broken unit tests