Forgot to amend test