Fixed bug from copy and paste code -_-