Fix tests
Merge pull request #14 from kevinwilfong/test

Fix tests