Fixed and included an incomplete unit test