Fixing unit test - last time