fixed test to work with builder