Adds additional return data to fix test failure