quick solution .. fixed