Update tests

Fixes Travis build for patch #3