Edited Unit test to get travis to run