Fixed a little bug
Fixed a little bug