fixing that bug