Added serialVersionUID. 
Fixed little bug.