Fix incorrect extension import in agent.

Change-Id: I2c9fef70e913bbdbda4a3bf255111bb03a9fe275