fix for minor bug