Fix for minor bug