Fixed using wrong class & flag.