Fix last build error