Fixed example test to pass