Small fix, removed test code change