Fixed classname bug