More fixes, cause I screwed up more
Add missing line
Merge pull request #12 from Trinovantes/fix-more-backend-logic

More fixes, cause I screwed up more