Fixed class

Don't know how to revert changes in one file. TODO: Learn how to.