fix ugly bug :-)