Fixes import error in class flags.

Probably due to an incorrect merge?