Add forgotten use statement