Fixes TestNG tests.