Fix for travis fail.

#2