Forgot this tiny modification