Remove uncessary test code