Promoting test to main class