Change the variable for primary key
Merge pull request #9 from Shine-neko/master

Fix primary key