Add another test. Should probably have checked more thoroughly