Fix Small Bug