Fix check, updated include