Fixed small bug

fixed crash