Fixed Small Bug