Fixed #3 for Peewee

Signed-off-by: Kaley Halloran <kaleyhalloran@gmail.com>