Added code injection for testing.