removed another bug in code