[FIX] Unit tests