Fix class.

Fix class.