Fixed lint in code