Update build file for unit tests