Testing now works properly