Fix one more API incompatibility change