slight fix to build