Update test

Lucien. 
I have made a change to your code here. Please review it and decide if you want to keep those changes.
Mk.