Working version of the test