updates unit tests to pass.