IM** - fix broken tests