Fixes minor issue introduced in previous commit