Shaft extensions with square profile