Small change in where the class ends.