Fix primary key + add test
Bump to 1.0.1