Fixed failing test

All tests should be green now