Small tweak to build files

Small tweak to build files