fix baby step 2
Merge pull request #7 from mohicanlove/fix-step-2

fix baby step 2