Minor edit to square mixin