Change to class name in order to make it all work properly.