fix that bug