Fix #11 "Use WP functions instead"