Update unit tests to use environment vars