fix broken tests (#194)