found bug and fixed it ... probably working