[TEST] Fixed unit test....