Fix build: missing import