Remove test code added in last commit