Fixing injection issue for testing purposes