Adding passing test
Fixing test