Fixing entity/behavor compilation error