Minor update to pass mvn test