small addition to fix propably issue