Add missing import (missed because not used)