Updated tests to make them pass