Just fixed the code "design"