Fixed simple bug