Change test to reflect interface expectations