fix of testing setup

The Done() was not in the right area.
Call me potatoehead for not being able to copy-paste code ^^...
Tests work now also for the first test case.