Allow to specify test class