fixed broken test... not well refactored