Fixed failing tests, silly mistake