Fixed unit test error
Updates