tests passing locally, failing in travis. fix #2