Minor tweak in controller