Fixed compilation error and removed unnecessary import statement. 

Fixed compilation error and removed unnecessary import statement.