Change assigment to constructor injection