Fixed small bug oversight.