Remove part of testing code