Small fix to test and POM