Fix jmx tests