Fix tests extending UnitSpec