Active display: Unify (un)registering callbacks and add proper checks.

Change-Id: Ie4c4660a54d2c60831f7a3bcc66b3a9ca1d035f2