Fix a crash introduced in last commit