Minor update for compatibility