Move class out of yield block