Added another unit test

Unit tests now running consistently