Adding injected instance
Fixed the test methods by changing the given section in all methods.