Fix tests

For some reason these were failing in my machine. I don't know why they were passing in Travis though. Any ideas?