Fixing tests.

#108