Correcting for error in primary beam calculation