Removed test code and modified a comment