Fixed error in tests.
Fucking hilights works