Fix build which was failing after last commit