Also required for #328

Forgot about this in the original commit