Updated __init__ to comply with inherited class.

Change-Id: Id355dbda59f326c010b5ba1428027322050e67a8