Fix problem with previous commit