Small modification for a bug