small change to fix base case