fixed #7
Tested with WordPress 3.7