static int ulite_console_setup(struct console *co, char *options)
 {
-       struct uart_port *port;
+       struct uart_port *port = NULL;
        int baud = 9600;
        int bits = 8;
        int parity = 'n';
        int flow = 'n';
 
-
-       port = console_port;
+       if (co->index >= 0 && co->index < ULITE_NR_UARTS)
+               port = ulite_ports + co->index;
 
        /* Has the device been initialized yet? */
-       if (!port->mapbase) {
+       if (!port || !port->mapbase) {
                pr_debug("console on ttyUL%i not present\n", co->index);
                return -ENODEV;
        }
 
+       console_port = port;
+
        /* not initialized yet? */
        if (!port->membase) {
                if (ulite_request_port(port))
 
        dev_set_drvdata(dev, port);
 
-#ifdef CONFIG_SERIAL_UARTLITE_CONSOLE
-       /*
-        * If console hasn't been found yet try to assign this port
-        * because it is required to be assigned for console setup function.
-        * If register_console() don't assign value, then console_port pointer
-        * is cleanup.
-        */
-       if (ulite_uart_driver.cons->index == -1)
-               console_port = port;
-#endif
-
        /* Register the port */
        rc = uart_add_one_port(&ulite_uart_driver, port);
        if (rc) {
                return rc;
        }
 
-#ifdef CONFIG_SERIAL_UARTLITE_CONSOLE
-       /* This is not port which is used for console that's why clean it up */
-       if (ulite_uart_driver.cons->index == -1)
-               console_port = NULL;
-#endif
-
        return 0;
 }