Update interface for new argument