Quick fix to small bug.