fixed unit test - fixes #363 again :)