Add missing import (oops).