Fix Unit tests

fixed unit tests to see if travis-ci works