Fixed contract classes changes during tests.