still trying to fix last bug. All other tests pass