move test to own class