Forgot to change the class names this should fix it