Updating POM and fixing unit test