Tests refactoring.
Tests refactoring.