Minor tests migration.
minor change