Fixing incorrect test and adding a new stub