Fixed test code after switching to Maven.