fixed an issue caused by last bug fix