Both are correct IMO