Fixed small bug in tests.
Small cosmetic fixes.