Moved conditional compile into Makefile

Signed-off-by: Andreas Engel <andreas.engel@ericsson.com>
diff --git a/drivers/serial/Makefile b/drivers/serial/Makefile
index 3cc1999..b370828 100644
--- a/drivers/serial/Makefile
+++ b/drivers/serial/Makefile
@@ -33,7 +33,8 @@
 COBJS-$(CONFIG_S3C64XX) += s3c64xx.o
 COBJS-y += serial.o
 COBJS-$(CONFIG_MAX3100_SERIAL) += serial_max3100.o
-COBJS-y += serial_pl01x.o
+COBJS-$(CONFIG_PL010_SERIAL) += serial_pl01x.o
+COBJS-$(CONFIG_PL011_SERIAL) += serial_pl01x.o
 COBJS-$(CONFIG_XILINX_UARTLITE) += serial_xuartlite.o
 COBJS-$(CONFIG_SCIF_CONSOLE) += serial_sh.o
 COBJS-$(CONFIG_USB_TTY) += usbtty.o
diff --git a/drivers/serial/serial_pl01x.c b/drivers/serial/serial_pl01x.c
index d0497ec..c645cef 100644
--- a/drivers/serial/serial_pl01x.c
+++ b/drivers/serial/serial_pl01x.c
@@ -25,14 +25,11 @@
  * MA 02111-1307 USA
  */
 
-/* Simple U-Boot driver for the PrimeCell PL011 UARTs on the IntegratorCP */
-/* Should be fairly simple to make it work with the PL010 as well */
+/* Simple U-Boot driver for the PrimeCell PL010/PL011 UARTs */
 
 #include <common.h>
 #include <watchdog.h>
 
-#if defined(CFG_PL010_SERIAL) || defined(CFG_PL011_SERIAL)
-
 #include "serial_pl01x.h"
 
 #define IO_WRITE(addr, val) (*(volatile unsigned int *)(addr) = (val))
@@ -52,7 +49,7 @@
 static int pl01x_getc (int portnum);
 static int pl01x_tstc (int portnum);
 
-#ifdef CFG_PL010_SERIAL
+#ifdef CONFIG_PL010_SERIAL
 
 int serial_init (void)
 {
@@ -110,9 +107,9 @@
 	return 0;
 }
 
-#endif /* CFG_PL010_SERIAL */
+#endif /* CONFIG_PL010_SERIAL */
 
-#ifdef CFG_PL011_SERIAL
+#ifdef CONFIG_PL011_SERIAL
 
 int serial_init (void)
 {
@@ -157,7 +154,7 @@
 	return 0;
 }
 
-#endif /* CFG_PL011_SERIAL */
+#endif /* CONFIG_PL011_SERIAL */
 
 void serial_putc (const char c)
 {
@@ -224,5 +221,3 @@
 	return !(IO_READ (port[portnum] + UART_PL01x_FR) &
 		 UART_PL01x_FR_RXFE);
 }
-
-#endif