fix error: cannot import name BUILDING_KEY