Update test

Resolved issue #2