Update test based on interface changes