Fix build for 4.01