Minor update
Fixed a bug.