Fix bug in squares controller
Squares bug fix with header