[Fix] Fixed small bug (shame on me...)