quick fix for a minor bug