Remove test code and fix copy paste error