Fixed tests for Square.