Fix build after fix for #5