fixed issue caused by last change