Fixed issue #8
Using pk instead of id