Fixed a little bug