Adding missing test