Fixed complilation errors