Fixed forgotten changes.