Fixed bug from end of class.