Tried using that way but it breaks the tests.