Convert to di from junitTest