Fix unit test

Needs to run with spring context