Updating tests to pass based on story model change