Remove unit test