forgot to update this class