Modified unit test to attempt to pass on Travis