Fix for small bug that causes a lot of warnings.