removing code from testing