Testing works properly now