Update classname to the correct one