Travis-CI fix

Resolved #2