Fixed bug from copy paste