Fix wrong code in unittest example.
Merge pull request #407 from AndrejMitrovic/FixupUnittest

Fix wrong code in unittest example.