Fix wrong header imports.

There'll be an error when I use it in Xcode 6, so I just fix it.