small addition to previous fix