Please make this an inner class of the test?