added 2 lines, fixes #1