Small update.

Had to fix a small bug.