Remove missed test code