Attempting to fix minor bug.