Fixed JUnit test after changes in EJB