Fixed bug due to previous changes.