Bug fix on controller