fix unit tests -- hopefully