Test refactoring

Ok.