Fixed mistake caused accidentally

Fixed the issue caused accidentally.