Fixed shitty TestNG things