Made changes so running the maven test work